import { Plugin } from 'vite/node_modules/rollup';

/**
 * 代码模板替换
 */
function plugin(config: {
    values: {
        [mapkey: string]: string;
    };
}): Plugin {
    return {
        name: 'rollup-plugin-replace',

        renderChunk(code) {
            for (const key in config.values) {
                const reg = new RegExp(`${key}`, 'g');
                code = code.replace(reg, config.values[key]);
            }
            return { code };
        },
    };
}

export default plugin;
